#include <stdio.h>
#include "front.h"
#include "admin.h"
#include "common.h"

void welcome();
int main(void) {
    open_db(); // 打开数据库连接
//    // 初始化数据库
//    if (init_db() == ERROR){
//        printf("数据库初始化失败!\n");
//        return 0;
//    }
    int choose;
    int flag = 0; // 退出标记 1退出
    while (1){
        welcome();
        scanf("%d",&choose);
        switch (choose) {
            case 1:
                // 选择用户登录，进入前台
                front_entry();
                break;
            case 2:
                admin_entry();
                break;
            case 3:
                // 注册功能
                registry();
                break;
            case 0:
                flag = 1;
                break;
            default:
                printf("输入错误,请重新输入!\n");
                break;
        }
        if (flag){
            printf("感谢您的使用!\n");
            close_db(); // 退出之前关闭数据库
            break;
        }
    }
    return 0;
}

void welcome(){
    printf("+*===========**--   超市购物管理系统   --**==========*+\n");
    printf("|**|  *-------------------------------------*   |**|\n");
    printf("|**|  |  - 1. 用户登录        - 2. 管理员登录  |   |**|\n");
    printf("|**|  |  - 3. 用户注册        - 0. 退出系统    |   |**|\n");
    printf("|**|  *-------------------------------------*   |**|\n");
    printf("+==================================================+\n");
    printf("请选择:");
    fflush(stdout);
}
